package com.mdnote.structure.treeapplication;

import org.omg.CORBA.DATA_CONVERSION;

import java.util.Arrays;

/**
 * @author Rhythm-2019
 * @version 1.0
 * @date 2020/8/30
 * @description 堆排序
 */
public class HeapSort {

    /**
     * 调整成大顶堆
     * @param source 数组
     * @param index 起始元素
     * @param length 长度
     */
    public static void adjustHeap(int[] source, int index, int length) {
        // 保存当前的值
        int temp = source[index];
        // 自下往上，调整
        // index * 2 + 1是index的左叶子节点
        for (int i = index * 2 + 1; i < length; i = i * 2 + 1) {
            // 判断左右子节点哪个大
            // i + 1 < source.length 没有右节点就不用走这个判断了
            if (i + 1 < length && source[i] < source[i + 1]) {
                i++;
            }
            // 找出父节点、左叶子节点和右子节点中最大的那个
            if (temp < source[i]) {
                source[index] = source[i];
                // 这里保存是为了最后把temp的值放回去
                index = i;
            } else {
                // 现在index所指向的值已经是最大的了
                break;
            }
        }
        //
        source[index] = temp;
    }

    private static void sort(int[] data) {
        // 从最后一个非叶子节点开始调整，调整完最大的元素已经在0位了
        for (int i = data.length / 2 - 1; i >= 0; i--) {
            adjustHeap(data, i, data.length);
        }
        // data.length - 1 表示最后一个元素不用去处理
        for (int i = data.length - 1; i > 0; i--) {
            int temp = data[0];
            data[0] = data[i];
            data[i] = temp;
            adjustHeap(data, 0, i);
        }
    }

    public static void main(String[] args) {
        int[] data = {4, 6, 8, 5, 9, 10, 13, 1, 3, 7, 18};
        sort(data);
        System.out.println(Arrays.toString(data));
    }

}
